#!/bin/bash

# - Author: jiruyi
# - Filename: run.sh
# - Date: Tue Oct  9 09:16:18 CST 2018

# - Description:
#   second fine tune the whole layers

set -e
WORK_PATH=$(dirname $(readlink -f $0))
TOOLS=$CAFFE_ROOT/build/tools
LOG=$WORK_PATH/logs/log-server-`date +%Y-%m-%d-%H-%M-%S`-device_$1.log


#stage_second
$TOOLS/caffe train \
    -solver "$WORK_PATH/prototxts/stage_second/solver.prototxt"  \
    -weights "$WORK_PATH/snapshots/fgvc_first_iter_60000.caffemodel" \
    -log_dir "$WORK_PATH/logs/" \
    -gpu $1 \
    2>&1 | tee -a $LOG

